package skill;

public class sortColors {
    // public void sortColors(int[] nums) {
    // //统计各个颜色的数目
    // int[] color=new int[3];
    // for(int num:nums){
    // color[num]++;
    // }
    // int red=color[0],index=0;
    // while (red>0) {
    // if (nums[index]==0) {
    // red--;
    // index++;
    // }else{
    // int start=index;
    // while (nums[start]!=0&&start<nums.length) {
    // start++;
    // }
    // nums[start]=nums[index];
    // nums[index]=0;
    // index++;
    // red--;
    // }
    // }
    // int white=color[1];
    // while (white>0) {
    // if (nums[index]==1) {
    // white--;
    // index++;
    // }else{
    // int start=index;
    // while (nums[start]!=1&&start<nums.length) {
    // start++;
    // }
    // nums[start]=nums[index];
    // nums[index]=1;
    // index++;
    // white--;
    // }
    // }
    // for(int num:nums){
    // System.out.println(num);
    // }
    // }
    public void sortColors(int[] nums) {
        // 双指针
        int p0 = 0, p2 = nums.length - 1;
        for (int i = 0; i <= p2; i++) {
            while (i<=p2&&nums[i] == 2) {
                nums[i] = nums[p2];
                nums[p2] = 2;
                p2--;
            }
            //不是else if的关系，是因为可能把一个0丢去前面了，如果此时index++，0就被跳过了
            if (nums[i] == 0) {
                nums[i] = nums[p0];
                nums[p0] = 0;
                p0++;
            }
        }
        for (int num : nums) {
            System.out.println(num);
        }
    }

    public static void main(String[] args) {
        sortColors obj = new sortColors();
        // obj.sortColors(new int[]{2,0,2,1,1,0});
        // obj.sortColors(new int[] { 2, 0, 1 });
        obj.sortColors(new int[] { 0,1,1,2,2, 0, 1 });
    }
}
